Node / ReticulumProjects / MeshChatX.git / files / tests / backend / test_ws_json_contracts.py

Displaying Raw • Download

T8b949e# SPDX-License-Identifier: 0BSD

T8b949e"""WebSocket JSON message contract tests."""

Tff7b72from T7ee787__future__ Tff7b72import Te6edf3annotations

Tff7b72import T7ee787json

Tff7b72from T7ee787pathlib Tff7b72import Te6edf3Path

Tff7b72from T7ee787unittestT7ee787.T7ee787mock Tff7b72import Te6edf3AsyncMockTb4b4b4, Te6edf3MagicMockTb4b4b4, Te6edf3patch

Tff7b72import T7ee787pytest

Tff7b72from T7ee787meshchatxT7ee787.T7ee787meshchat Tff7b72import Te6edf3ReticulumMeshChat

Tff7b72from T7ee787testsT7ee787.T7ee787backendT7ee787.T7ee787ws_contract_helpers Tff7b72import Tb4b4b4(

Te6edf3extract_client_direct_response_typesTb4b4b4,

Te6edf3extract_client_inbound_typesTb4b4b4,

Te6edf3extract_server_broadcast_typesTb4b4b4,

Te6edf3load_ws_manifestTb4b4b4,

Te6edf3write_ws_manifestTb4b4b4,

Tb4b4b4)

Tff7b72from T7ee787testsT7ee787.T7ee787backendT7ee787.T7ee787ws_json_contract_schemas Tff7b72import Tb4b4b4(

Te6edf3WS_MESSAGE_SAMPLESTb4b4b4,

Te6edf3WS_MESSAGE_SCHEMASTb4b4b4,

Te6edf3all_ws_message_typesTb4b4b4,

Te6edf3assert_ws_message_matches_schemaTb4b4b4,

Tb4b4b4)

Te6edf3_REPO_ROOT Tff7b72= Te6edf3PathTb4b4b4(Tff7b72__file__Tb4b4b4)Tff7b72.Td2a8ffresolveTb4b4b4(Tb4b4b4)Tff7b72.Td2a8ffparentsTb4b4b4[T79c0ff2Tb4b4b4]

Te6edf3_MESHCHAT_PY Tff7b72= Te6edf3_REPO_ROOT Tff7b72/ Ta5d6ff"Ta5d6ffmeshchatxTa5d6ff" Tff7b72/ Ta5d6ff"Ta5d6ffmeshchat.pyTa5d6ff"

Te6edf3_FIXTURE Tff7b72= Te6edf3PathTb4b4b4(Tff7b72__file__Tb4b4b4)Tff7b72.Td2a8ffresolveTb4b4b4(Tb4b4b4)Tff7b72.Td2a8ffparent Tff7b72/ Ta5d6ff"Ta5d6fffixturesTa5d6ff" Tff7b72/ Ta5d6ff"Ta5d6ffws_message_manifest.jsonTa5d6ff"

Tff7b72def Td2a8ff_all_manifest_typesTb4b4b4(Te6edf3manifestTb4b4b4: Tffa657dictTb4b4b4[Tffa657strTb4b4b4, Tffa657listTb4b4b4[Tffa657strTb4b4b4]Tb4b4b4]Tb4b4b4) Tff7b72-Tff7b72> Tffa657setTb4b4b4[Tffa657strTb4b4b4]Tb4b4b4:

Te6edf3outTb4b4b4: Tffa657setTb4b4b4[Tffa657strTb4b4b4] Tff7b72= Tffa657setTb4b4b4(Tb4b4b4)

Tff7b72for Te6edf3key Tff7b72in Tb4b4b4(Ta5d6ff"Ta5d6ffclient_inboundTa5d6ff"Tb4b4b4, Ta5d6ff"Ta5d6ffclient_direct_responsesTa5d6ff"Tb4b4b4, Ta5d6ff"Ta5d6ffserver_broadcastTa5d6ff"Tb4b4b4)Tb4b4b4:

Te6edf3outTff7b72.Td2a8ffupdateTb4b4b4(Te6edf3manifestTff7b72.Td2a8ffgetTb4b4b4(Te6edf3keyTb4b4b4, Tb4b4b4[Tb4b4b4]Tb4b4b4)Tb4b4b4)

Tff7b72return Te6edf3out

Tff7b72def Td2a8fftest_ws_message_manifest_matches_meshchatTb4b4b4(Tb4b4b4)Tb4b4b4:

Te6edf3live Tff7b72= Tb4b4b4{

Ta5d6ff"Ta5d6ffclient_inboundTa5d6ff"Tb4b4b4: Te6edf3extract_client_inbound_typesTb4b4b4(Te6edf3_MESHCHAT_PYTb4b4b4)Tb4b4b4,

Ta5d6ff"Ta5d6ffclient_direct_responsesTa5d6ff"Tb4b4b4: Te6edf3extract_client_direct_response_typesTb4b4b4(Te6edf3_MESHCHAT_PYTb4b4b4)Tb4b4b4,

Ta5d6ff"Ta5d6ffserver_broadcastTa5d6ff"Tb4b4b4: Te6edf3extract_server_broadcast_typesTb4b4b4(Te6edf3_MESHCHAT_PYTb4b4b4)Tb4b4b4,

Tb4b4b4}

Tff7b72if Tffa657__import__Tb4b4b4(Ta5d6ff"Ta5d6ffosTa5d6ff"Tb4b4b4)Tff7b72.Td2a8ffenvironTff7b72.Td2a8ffgetTb4b4b4(Ta5d6ff"Ta5d6ffUPDATE_WS_MESSAGE_MANIFESTTa5d6ff"Tb4b4b4) Tff7b72== Ta5d6ff"Ta5d6ff1Ta5d6ff"Tb4b4b4:

Te6edf3write_ws_manifestTb4b4b4(Te6edf3_FIXTURETb4b4b4, Te6edf3liveTb4b4b4)

Te6edf3pytestTff7b72.Td2a8ffskipTb4b4b4(

Ta5d6ff"Ta5d6ffUPDATE_WS_MESSAGE_MANIFEST=1: fixture updated; re-run without the env varTa5d6ff"Tb4b4b4,

Tb4b4b4)

Te6edf3expected Tff7b72= Te6edf3load_ws_manifestTb4b4b4(Te6edf3_FIXTURETb4b4b4)

Tff7b72assert Te6edf3live Tff7b72== Te6edf3expectedTb4b4b4, Tb4b4b4(

Ta5d6ff"Ta5d6ffWebSocket message manifest drifted. Run: Ta5d6ff"

Ta5d6ff"Ta5d6ffUPDATE_WS_MESSAGE_MANIFEST=1 uv run pytest Ta5d6ff"

Ta5d6ff"Ta5d6fftests/backend/test_ws_json_contracts.py -k manifest_matches_meshchatTa5d6ff"

Tb4b4b4)

Tff7b72def Td2a8fftest_every_manifest_type_has_schemaTb4b4b4(Tb4b4b4)Tb4b4b4:

Te6edf3manifest Tff7b72= Te6edf3load_ws_manifestTb4b4b4(Te6edf3_FIXTURETb4b4b4)

Te6edf3missing Tff7b72= Tffa657sortedTb4b4b4(Te6edf3_all_manifest_typesTb4b4b4(Te6edf3manifestTb4b4b4) Tff7b72- Te6edf3all_ws_message_typesTb4b4b4(Tb4b4b4)Tb4b4b4)

Tff7b72assert Tff7b72not Te6edf3missingTb4b4b4, Ta5d6fffTa5d6ff"Ta5d6ffMissing WebSocket schemas for: Tffd700{Te6edf3missingTffd700}Ta5d6ff"

Tff7b72def Td2a8fftest_every_schema_has_sample_payloadTb4b4b4(Tb4b4b4)Tb4b4b4:

Te6edf3missing Tff7b72= Tffa657sortedTb4b4b4(Te6edf3all_ws_message_typesTb4b4b4(Tb4b4b4) Tff7b72- Tffa657setTb4b4b4(Te6edf3WS_MESSAGE_SAMPLESTb4b4b4)Tb4b4b4)

Tff7b72assert Tff7b72not Te6edf3missingTb4b4b4, Ta5d6fffTa5d6ff"Ta5d6ffMissing WebSocket sample payloads for: Tffd700{Te6edf3missingTffd700}Ta5d6ff"

Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffparametrizeTb4b4b4(Ta5d6ff"Ta5d6ffmsg_typeTa5d6ff"Tb4b4b4, Tffa657sortedTb4b4b4(Te6edf3WS_MESSAGE_SCHEMASTb4b4b4)Tb4b4b4)

Tff7b72def Td2a8fftest_ws_schema_accepts_documented_sampleTb4b4b4(Te6edf3msg_typeTb4b4b4: Tffa657strTb4b4b4)Tb4b4b4:

Te6edf3sample Tff7b72= Te6edf3WS_MESSAGE_SAMPLESTb4b4b4[Te6edf3msg_typeTb4b4b4]

Tff7b72assert Te6edf3sampleTb4b4b4[Ta5d6ff"Ta5d6fftypeTa5d6ff"Tb4b4b4] Tff7b72== Te6edf3msg_type

Te6edf3assert_ws_message_matches_schemaTb4b4b4(Te6edf3sampleTb4b4b4)

Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffasyncio

Tff7b72async Tff7b72def Td2a8fftest_ws_ping_emits_pong_contractTb4b4b4(Tb4b4b4)Tb4b4b4:

Te6edf3app Tff7b72= Te6edf3ReticulumMeshChatTff7b72.Td2a8ff__new__Tb4b4b4(Te6edf3ReticulumMeshChatTb4b4b4)

Te6edf3client Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

Te6edf3sentTb4b4b4: Tffa657listTb4b4b4[Tffa657strTb4b4b4] Tff7b72= Tb4b4b4[Tb4b4b4]

Tff7b72async Tff7b72def Td2a8ffcapture_sendTb4b4b4(Te6edf3payloadTb4b4b4: Tffa657strTb4b4b4)Tb4b4b4:

Te6edf3sentTff7b72.Td2a8ffappendTb4b4b4(Te6edf3payloadTb4b4b4)

Te6edf3clientTff7b72.Td2a8ffsend_str Tff7b72= Te6edf3AsyncMockTb4b4b4(Te6edf3side_effectTff7b72=Te6edf3capture_sendTb4b4b4)

Tff7b72with Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.meshchat.AsyncUtilsTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3mock_async_utilsTb4b4b4:

Tff7b72def Td2a8ffrun_asyncTb4b4b4(Te6edf3coroTb4b4b4)Tb4b4b4:

Tff7b72import T7ee787asyncio

Tff7b72if Te6edf3asyncioTff7b72.Td2a8ffiscoroutineTb4b4b4(Te6edf3coroTb4b4b4)Tb4b4b4:

Te6edf3asyncioTff7b72.Td2a8ffget_event_loopTb4b4b4(Tb4b4b4)Tff7b72.Td2a8ffcreate_taskTb4b4b4(Te6edf3coroTb4b4b4)

Te6edf3mock_async_utilsTff7b72.Td2a8ffrun_asyncTff7b72.Td2a8ffside_effect Tff7b72= Te6edf3run_async

Tff7b72await Te6edf3appTff7b72.Td2a8ffon_websocket_data_receivedTb4b4b4(Te6edf3clientTb4b4b4, Tb4b4b4{Ta5d6ff"Ta5d6fftypeTa5d6ff"Tb4b4b4: Ta5d6ff"Ta5d6ffpingTa5d6ff"Tb4b4b4}Tb4b4b4)

Tff7b72for Te6edf3payload Tff7b72in Te6edf3sentTb4b4b4:

Te6edf3assert_ws_message_matches_schemaTb4b4b4(Te6edf3jsonTff7b72.Td2a8ffloadsTb4b4b4(Te6edf3payloadTb4b4b4)Tb4b4b4)

Tff7b72assert Te6edf3jsonTff7b72.Td2a8ffloadsTb4b4b4(Te6edf3payloadTb4b4b4)Tb4b4b4[Ta5d6ff"Ta5d6fftypeTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6ffpongTa5d6ff"